"""
Copyright (c) Facebook, Inc. and its affiliates.
This source code is licensed under the MIT license found in the
LICENSE file in the root directory of this source tree.
"""

import logging

def get_config(name):

    config = {}

    if name.upper() == 'UCF101':
        config['num_classes'] = 101
    elif name.upper() == 'HMDB51':
        config['num_classes'] = 51
    else:
        logging.error("Configs for dataset '{}'' not found".format(name))
        raise NotImplemented

    logging.debug("Target dataset: '{}', configs: {}".format(name.upper(), config))

    return config


if __name__ == "__main__":
    logging.getLogger().setLevel(logging.DEBUG)

    logging.info(get_config("ucf101"))
    logging.info(get_config("HMDB51"))
